// Copyright 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "cc/resources/ui_resource_bitmap.h"

#include <stdint.h>

#include <memory>

#include "base/logging.h"
#include "third_party/skia/include/core/SkImageInfo.h"
#include "third_party/skia/include/core/SkMallocPixelRef.h"
#include "third_party/skia/include/core/SkPixelRef.h"

namespace cc {
namespace {

    UIResourceBitmap::UIResourceFormat SkColorTypeToUIResourceFormat(
        SkColorType sk_type)
    {
        UIResourceBitmap::UIResourceFormat format = UIResourceBitmap::RGBA8;
        switch (sk_type) {
        case kN32_SkColorType:
            format = UIResourceBitmap::RGBA8;
            break;
        case kAlpha_8_SkColorType:
            format = UIResourceBitmap::ALPHA_8;
            break;
        default:
            NOTREACHED() << "Invalid SkColorType for UIResourceBitmap: " << sk_type;
            break;
        }
        return format;
    }

} // namespace

void UIResourceBitmap::Create(sk_sp<SkPixelRef> pixel_ref,
    const gfx::Size& size,
    UIResourceFormat format)
{
    DCHECK(size.width());
    DCHECK(size.height());
    DCHECK(pixel_ref);
    DCHECK(pixel_ref->isImmutable());
    format_ = format;
    size_ = size;
    pixel_ref_ = std::move(pixel_ref);

    // Default values for secondary parameters.
    opaque_ = (format == ETC1);
}

void UIResourceBitmap::DrawToCanvas(SkCanvas* canvas, SkPaint* paint)
{
    SkBitmap bitmap;
    bitmap.setInfo(pixel_ref_.get()->info(), pixel_ref_.get()->rowBytes());
    bitmap.setPixelRef(pixel_ref_.get());
    canvas->drawBitmap(bitmap, 0, 0, paint);
    canvas->flush();
}

UIResourceBitmap::UIResourceBitmap(const SkBitmap& skbitmap)
{
    DCHECK_EQ(skbitmap.width(), skbitmap.rowBytesAsPixels());
    DCHECK(skbitmap.isImmutable());

    sk_sp<SkPixelRef> pixel_ref = sk_ref_sp(skbitmap.pixelRef());
    const SkImageInfo& info = pixel_ref->info();
    Create(std::move(pixel_ref), gfx::Size(info.width(), info.height()),
        SkColorTypeToUIResourceFormat(skbitmap.colorType()));

    SetOpaque(skbitmap.isOpaque());
}

UIResourceBitmap::UIResourceBitmap(const gfx::Size& size, bool is_opaque)
{
    SkAlphaType alphaType = is_opaque ? kOpaque_SkAlphaType : kPremul_SkAlphaType;
    SkImageInfo info = SkImageInfo::MakeN32(size.width(), size.height(), alphaType);
    sk_sp<SkPixelRef> pixel_ref(
        SkMallocPixelRef::NewAllocate(info, info.minRowBytes(), NULL));
    pixel_ref->setImmutable();
    Create(std::move(pixel_ref), size, UIResourceBitmap::RGBA8);
    SetOpaque(is_opaque);
}

UIResourceBitmap::UIResourceBitmap(sk_sp<SkPixelRef> pixel_ref,
    const gfx::Size& size)
{
    Create(std::move(pixel_ref), size, UIResourceBitmap::ETC1);
}

UIResourceBitmap::UIResourceBitmap(const UIResourceBitmap& other) = default;

UIResourceBitmap::~UIResourceBitmap() { }

AutoLockUIResourceBitmap::AutoLockUIResourceBitmap(
    const UIResourceBitmap& bitmap)
    : bitmap_(bitmap)
{
    bitmap_.pixel_ref_->lockPixels();
}

AutoLockUIResourceBitmap::~AutoLockUIResourceBitmap()
{
    bitmap_.pixel_ref_->unlockPixels();
}

const uint8_t* AutoLockUIResourceBitmap::GetPixels() const
{
    return static_cast<const uint8_t*>(bitmap_.pixel_ref_->pixels());
}

} // namespace cc
